原文題目
Given the root
of a binary tree, invert the tree, and return its root.
題目摘要
root
Example 1:
Input: root = [4,2,7,1,3,6,9]
Output: [4,7,2,9,6,3,1]
Example 2:
Input: root = [2,1,3]
Output: [2,3,1]
Example 3:
Input: root = []
Output: []
解題思路
初步檢查:
如果當前節點為 null
,這意味著已經到達樹的末端或者樹是空的,因此直接返回 null
。
反轉當前節點的左右子樹:
temp
來存儲當前節點的左子樹。temp
(即原來的左子樹)。遞迴處理子樹:
遞迴使用 invertTree
方法分別對反轉後的左子樹和右子樹進行相同的操作。
回傳反轉後的根節點:
完成當前節點及其子樹的反轉後,則回傳反轉後的根節點。
程式碼
class Solution {
public TreeNode invertTree(TreeNode root) {
//如果當前節點為null則返回
if (root == null) {
return null;
}
//設立temp來輔助左、右子樹的質做交換
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
invertTree(root.left); //遞迴反轉左子樹
invertTree(root.right); //遞迴反轉右子樹
return root; //反轉完成則回傳反轉後的根節點
}
}
結論
我覺得這題的解法非常直觀,只需要進行左右子樹的交換,再透過遞迴遍歷整棵樹,就能得到正解,過程中只要確保終止條件和左右子樹的交換順序無誤,整個流程就能順利執行啦!